TRAINING_PARAMS = \
{
    "model_params": {
        "backbone_name": "darknet_53",
        "backbone_pretrained": "../weights/darknet53_weights_pytorch.pth",  # set empty to disable
    },
    "yolo": {
        # 原始数据集的anchors
        # "anchors": [[[12, 15], [12, 13], [14, 17]],
        #             [[10, 16], [11, 13], [12, 18]],
        #             [[9, 10], [10, 14], [10, 12]]],
        # "classes": 1,
        # oblique_U_hoop_nut_06
        # "anchors": [[[14, 16], [15, 17], [16, 21]],
        #             [[13, 17], [13, 13], [14, 19]],
        #             [[10, 13], [11, 17], [11, 15]]],
        # "classes": 80,
        # oblique_U_hoop_nut_23
        # "anchors": [[[12, 14], [13, 17], [15, 21]],
        #             [[11, 16], [12, 21], [12, 19]],
        #             [[10, 14], [10, 20], [10, 17]]],
        # "classes": 80,
        # flat_U_hoop_nut_22
        # "anchors": [[[13, 21], [14, 18], [15, 22]],
        #             [[11, 19], [11, 13], [13, 15]],
        #             [[9, 12], [10, 14], [11, 16]]],
        # "classes": 80,
        # flat_U_hoop_nut_05
        "anchors": [[[12, 15], [13, 18], [14, 15]],
                    [[11, 17], [11, 14], [12, 12]],
                    [[9, 10], [10, 15], [10, 12]]],
        "classes": 80,
    },
    "lr": {
        "backbone_lr": 0.001,
        "other_lr": 0.01,
        "freeze_backbone": False,   # freeze backbone wegiths to finetune
        "decay_gamma": 0.1,
        "decay_step": 20,           # decay lr in every ? epochs
    },
    "optimizer": {
        "type": "sgd",
        "weight_decay": 4e-05,
    },
    "batch_size": 16,
    "train_path": "/mnt/sda1/cq/4c/flat_U_hoop_nut_05/flat_05_image_416_sorted_train.txt",
    "train_label_path": "/mnt/sda1/cq/4c/flat_U_hoop_nut_05/label_normalize/flat_05_label_all_sorted_train.txt",
    "epochs": 120,
    "img_h": 416,
    "img_w": 416,
    "parallels": [0, 2],                             # config GPU device
    "working_dir": "../bobliu_yolov3",      # replace with your working dir
    "pretrain_snapshot": "",    # load checkpoint
    "evaluate_type": "",
    "try": 4,
    "export_onnx": False,
}
